'''
v1.17
        新增功能
        1.爆炸效果实现
'''
import pygame,time,random
_display = pygame.display
version="v1.17"
class MainGame():
    #游戏主窗口
    window = None
    SCREEN_HEIGHT = 800
    SCREEN_WIDTH = 1200
    COLOR_BLACK=pygame.Color(0,0,0)
    #创建我方坦克
    TANK_P1 = None
    EnemyTank_list = []
    EnemyTank_count = 8
    #存储我方子弹的列表
    Bullet_list = []
    #存储敌方子弹列表
    Enemy_bullet_list = []
    #爆炸效果列表
    Explode_list = []
    def __init__(self):
        pass
    #开始游戏方法
    def startGame(self):
        _display.init()
        #创建窗口加载窗口
        MainGame.window = _display.set_mode(size=(1200,800))
        #创建我方坦克
        MainGame.TANK_P1=Tank(600,750)
        #？创建敌方坦克
        self.creatEnemyTank()
        #设置游戏标题
        _display.set_caption("坦克大战"+version)
        # self.getTextSurface("aa")
        #让窗口出口持续刷新操作
        while True:
            #给窗口填充颜色
            MainGame.window.fill(MainGame.COLOR_BLACK)
            #在循环中持续完成事件获取
            self.getEvent()
            #将绘制文字得到的画布，粘贴到窗口
            MainGame.window.blit(self.getTextSurface("剩余地方坦克%d辆"%len(MainGame.EnemyTank_list)),(5,5))
            #将我防坦克加入窗口
            MainGame.TANK_P1.displayTank()
            #展示敌方坦克
            self.biltEnemyTank()
            #调用坦克移动方法
            if MainGame.TANK_P1 and not MainGame.TANK_P1.stop:
                MainGame.TANK_P1.move()
            #调用渲染子弹列表的方法
            self.blitBullet()
            # 调用渲染敌方子弹列表的方法
            self.blitEnemyBullet()
            #调用爆炸效果
            self.displayExplodes()
            time.sleep(0.02)
            #刷新窗口
            _display.update()
    #创建敌方坦克
    def creatEnemyTank(self):
        top = 100
        for i in range(MainGame.EnemyTank_count):
            speed = random.randint(2, 6)
            left = random.randint(1, 11)
            eTank = EnemyTank(left*100,top,speed)
            MainGame.EnemyTank_list.append(eTank)
    #将difang坦克加入窗口
    def biltEnemyTank(self):
        for eTank in MainGame.EnemyTank_list:
            if eTank.live:
                eTank.displayTank()
                #敌方坦克移动
                eTank.randMove()
                #敌方坦克射击
                eBullet = eTank.shot()
                if eBullet:
                #将子弹存储到敌方子弹列表中
                    MainGame.Enemy_bullet_list.append(eBullet)
            else:
                MainGame.EnemyTank_list.remove(eTank)
    def blitBullet(self):
        for bullet in MainGame.Bullet_list:
            #如果子弹还活着，绘制出来，否则，直接从列表中移除子弹
            if bullet.live:
                bullet.displayBullet()
                bullet.bulletMove()
                #调用我方子弹与敌方坦克的碰撞方法
                bullet.hitEnemyTank()
            else:
                MainGame.Bullet_list.remove(bullet)
    #将敌方子弹加入到窗口中
    def blitEnemyBullet(self):
        for eBullet in MainGame.Enemy_bullet_list:
            #如果子弹还活着，绘制出来，否则，直接从列表中移除子弹
            if eBullet.live:
                eBullet.displayBullet()
                eBullet.bulletMove()
            else:
                MainGame.Enemy_bullet_list.remove(eBullet)
    #展示爆炸效果
    def displayExplodes(self):
        for explode in MainGame.Explode_list:
            if explode.live:
                explode.displayExplode()
            else:
                MainGame.Explode_list.remove(explode)
    #获取程序期间所有事件（鼠标事件，键盘事件）
    def getEvent(self):
        #获取所有事件
        eventList=pygame.event.get()
        #1,点击关闭结束 2.按下键盘上某个按钮
        for event in eventList:
            #p判断event.type 是否Quit，如果是退出，直接调用结束方法
            if event.type==pygame.QUIT:
                self.endGame()
            #判断事件类型是否为按键按下，如果是，继续判断按键是哪一个按键，进行对应处理
            if event.type==pygame.KEYDOWN:
                #具体是哪一个按键的处理
                if event.key==pygame.K_LEFT:
                    print("坦克向左调头，移动")
                    #修改坦克方向
                    MainGame.TANK_P1.direction="L"
                    MainGame.TANK_P1.stop=False
                elif event.key==pygame.K_RIGHT:
                    print("坦克向右调头，移动")
                    # 修改坦克方向
                    MainGame.TANK_P1.direction = "R"
                    MainGame.TANK_P1.stop=False
                elif event.key == pygame.K_DOWN:
                    print("坦克向下调头，移动")
                    # 修改坦克方向
                    MainGame.TANK_P1.direction = "D"
                    MainGame.TANK_P1.stop=False
                elif event.key == pygame.K_UP:
                    print("坦克向上调头，移动")
                    # 修改坦克方向
                    MainGame.TANK_P1.direction = "U"
                    MainGame.TANK_P1.stop=False
                elif event.key == pygame.K_SPACE:
                    print("发射子弹")
                    if len(MainGame.Bullet_list) < 5:
                        #产生子弹
                        m = Bullet(MainGame.TANK_P1)
                        #将子弹加入到子弹列表
                        MainGame.Bullet_list.append(m)
                    else:
                        print("屏幕子弹饱和")
            if event.type==pygame.KEYUP:
                #修改坦克的移动状态
                if event.key==pygame.K_LEFT or event.key==pygame.K_RIGHT\
                    or event.key==pygame.K_UP or event.key==pygame.K_DOWN:
                    MainGame.TANK_P1.stop=True
    #左上角文字绘制功能
    def getTextSurface(self,text):
        #初始化字体模块
        pygame.font.init()
        #选择合适字体
        # fontList=pygame.font.get_fonts()
        # print(fontList)
        font=pygame.font.SysFont("kaiti", 18,)
        #绘制
        textSurface=font.render(text,True,(255,0,0))
        return textSurface
    #结束游戏方法
    def endGame(self):
        print("谢谢使用")
        #结束python解释器
        exit()
class BassItem(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
class Tank(BassItem):
    def __init__(self,left,top):
        self.images={
            "U":pygame.image.load("img\images\WU.gif"),
            "D":pygame.image.load("img\images\WD.gif"),
            "L":pygame.image.load("img\images\WL.gif"),
            "R":pygame.image.load("img\images\WR.gif"),
        }
        self.direction="U"
        self.image=self.images[self.direction]
        #坦克所在区域
        self.rect =self.image.get_rect()
        #坦克距离x，y的位置
        self.rect.left=left
        self.rect.top=top
        #速度属性
        self.speed=5
        #新增属性：移动开关
        self.stop=True
        #新增属性live记录坦克是否活着
        self.live = True
    #坦克移动
    def move(self):
        if self.direction=="L":
            if self.rect.left>0:
                self.rect.left-=self.speed
        elif self.direction=="R":
            if self.rect.left<1150:
                self.rect.left+=self.speed
        elif self.direction=="U":
            if self.rect.top>0:
                self.rect.top-=self.speed
        elif self.direction=="D":
            if self.rect.top<750:
                self.rect.top+=self.speed
    #射击方法
    def shot(self):
        return Bullet(self)

    #展示坦克
    def displayTank(self):
        # 重新设置坦克的图片
        self.image=self.images[self.direction]
        #将坦克加入到窗口
        MainGame.window.blit(self.image,self.rect)
class MyTank(Tank):
    def __init__(self):
        pass
class EnemyTank(Tank):
    def __init__(self,left,top,speed):
        # super(EnemyTank,self).__init__(left,top)
        self.live = True
        self.images = {
            "U": pygame.image.load("img\images\EU.gif"),
            "D": pygame.image.load("img\images\ED.gif"),
            "L": pygame.image.load("img\images\EL.gif"),
            "R": pygame.image.load("img\images\ER.gif"),
        }
        self.direction = self.randDirection()
        self.image = self.images[self.direction]
        # 坦克所在区域
        self.rect = self.image.get_rect()
        # 坦克距离x，y的位置
        self.rect.left = left
        self.rect.top = top
        # 速度属性
        self.speed = speed
        # 新增属性：移动开关
        self.stop = True
        # 新增步数属性
        self.step=35
    def randDirection(self):
        num = random.randint(1,4)
        if num == 1:
            return "U"
        elif num == 2:
            return "D"
        elif num == 3:
            return "L"
        elif num == 4:
            return "R"
    # def displayEnemyTank(self):
    #     super().displayTank()
    #随机移动
    def randMove(self):
        if self.step <= 0:
            self.direction = self.randDirection()
            self.step = 30
        else:
            self.move()
            self.step -= 1
    def shot(self):
        num = random.randint(1,100)
        if num == 1:
            return Bullet(self)
class Bullet(BassItem):
    def __init__(self,tank):
        #图片
        self.images = {
            "U": pygame.image.load("img\images\MU.gif"),
            "D": pygame.image.load("img\images\MD.gif"),
            "L": pygame.image.load("img\images\ML.gif"),
            "R": pygame.image.load("img\images\MR.gif"),
        }
        # 方向
        self.direction = tank.direction
        self.image = self.images[self.direction]
        #位置
        self.rect = self.image.get_rect()
        if self.direction == "U":
            self.rect.left = tank.rect.left + tank.rect.width/2-self.rect.width/2
            self.rect.top = tank.rect.top - self.rect.height
        if self.direction == "D":
            self.rect.left = tank.rect.left + tank.rect.width / 2 - self.rect.width / 2
            self.rect.top = tank.rect.top + tank.rect.height
        if self.direction == "L":
            self.rect.left = tank.rect.left - self.rect.width
            self.rect.top = tank.rect.top + tank.rect.width/2 - self.rect.width/2
        if self.direction == "R":
            self.rect.left = tank.rect.left + tank.rect.width
            self.rect.top = tank.rect.top + tank.rect.width/2 - self.rect.width/2
        #速度
        self.speed = 7
        #记录子弹是否碰撞
        self.live = True
    #子弹移动
    def bulletMove(self):
        if self.direction == "U":
            if self.rect.top > 0:
                self.rect.top -= self.speed*1.5
            else:
                self.live = False
        elif self.direction == "D":
            if self.rect.top < MainGame.SCREEN_HEIGHT - self.rect.height:
                self.rect.top += self.speed*1.5
            else:
                self.live = False
        elif self.direction == "L":
            if self.rect.left > 0:
                self.rect.left -= self.speed*1.5
            else:
                self.live = False
        elif self.direction == "R":
            if self.rect.left < MainGame.SCREEN_WIDTH - self.rect.width:
                self.rect.left += self.speed*1.5
            else:
                self.live = False
    #子弹展示
    def displayBullet(self):
        # 重新设置子弹的图片
        self.image = self.images[self.direction]
        # 将子弹加入到窗口
        MainGame.window.blit(self.image, self.rect)
    #新增我方子弹碰撞敌方坦克的方法
    def hitEnemyTank(self):
        for eTank in MainGame.EnemyTank_list:
            if pygame.sprite.collide_rect(eTank,self):
                #产生爆炸效果
                explode = Explode(eTank)
                #将爆炸效果加入发入列表
                MainGame.Explode_list.append(explode)
                self.live = False
                eTank.live = False
class Explode():
    def __init__(self,tank):
        self.rect = tank.rect
        self.step = 0
        self.images = [
            pygame.image.load("img/images/0.gif"),
            pygame.image.load("img/images/1.gif"),
            pygame.image.load("img/images/2.gif"),
            pygame.image.load("img/images/3.gif"),
            pygame.image.load("img/images/4.gif"),
            pygame.image.load("img/images/5.gif"),
            pygame.image.load("img/images/6.gif"),
            pygame.image.load("img/images/7.gif"),
        ]
        self.image = self.images[self.step]
        self.live = True
    #展示爆炸效果
    def displayExplode(self):
        if self.step < len(self.images):
            MainGame.window.blit(self.image,self.rect)
            self.image = self.images[self.step]
            self.step += 1
        else:
            self.live = False
            self.step = 0
class Wall():
    def __init__(self):
        pass
    #展示墙壁的方法
    def displayWall(self):
        pass
class Music():
    def __init__(self):
        pass
    #开始播放音乐
    def play(self):
        pass
MainGame().startGame()